package com.ruoyi.business.tools;

import com.aspose.cells.License;
import com.aspose.cells.PdfCompliance;
import com.aspose.cells.PdfSaveOptions;
import com.aspose.cells.Workbook;
import java.io.*;
import java.util.List;

public class ExcelToPdf {
    /**
     * 获取license 去除水印
     *
     * @return
     */
    public static boolean getLicense() {
        boolean result = false;
        try {
            InputStream is = ExcelToPdf.class.getClassLoader().getResourceAsStream("\\license.xml");
            License aposeLic = new License();
            aposeLic.setLicense(is);
            result = true;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }

    /**
     * excel 转为pdf 输出。
     *
     * @param sourceFilePath excel文件
     * @param desFilePathd   pad 输出文件目录
     */
    public static void excel2pdf(String sourceFilePath, String desFilePathd, int showSheets) {
        if (!getLicense()) { // 验证License 若不验证则转化出的pdf文档会有水印产生
            return;
        }
        try {
            Workbook wb = new Workbook(sourceFilePath);// 原始excel路径

            FileOutputStream fileOS = new FileOutputStream(desFilePathd);
            PdfSaveOptions pdfSaveOptions = new PdfSaveOptions();
            pdfSaveOptions.setOnePagePerSheet(true);//设置合规性类型
            pdfSaveOptions.setMergeAreas(true);


            //int[] autoDrawSheets = {3};
            //当excel中对应的sheet页宽度太大时，在PDF中会拆断并分页。此处等比缩放。
            //autoDraw(wb,autoDrawSheets);

            //隐藏workbook中不需要的sheet页。
            printSheetPage(wb, showSheets);
            wb.save(fileOS, pdfSaveOptions);
            fileOS.flush();
            fileOS.close();
            System.out.println("完毕");

        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 选择输出的的sheet页。
     *
     * @param wb
     * @param page 显示页的sheet数组
     */
    public static void printSheetPage(Workbook wb, int page) {
//        for (int i = 1; i < wb.getWorksheets().getCount(); i++) {
//            wb.getWorksheets().get(i).setVisible(false);
//        }
        if (page == 0) {
            wb.getWorksheets().get(0).setVisible(true);
        } else {
            for (int i = 0; i < wb.getWorksheets().getCount(); i++) {
                wb.getWorksheets().get(i).setVisible(true);
                System.out.println("第" + i + "页");
            }
        }
    }
}
